package generator

/**
这份文件中的随机生成用于测试使用，不会用在主代码中
*/
import (
	"math/rand"
	"time"
)

func init() {
	rand.New(rand.NewSource(time.Now().UnixNano()))
}

func GenRandomUeName(ueType string) string {
	macbookName := []string{"Macbook Pro", "Macbook Air"}
	lenovoName := []string{"ThinkPad X1 Carbon", "ThinkBook 16", "ThinkPad T16"}
	dellName := []string{"Inspiron", "Vostro", "Alienware"}
	huaweiName := []string{"MateBook14", "MateBookX"}

	if ueType == "Macbook" {
		return macbookName[rand.Intn(len(macbookName))]
	} else if ueType == "Lenovo" {
		return lenovoName[rand.Intn(len(lenovoName))]
	} else if ueType == "Dell" {
		return dellName[rand.Intn(len(dellName))]
	} else {
		return huaweiName[rand.Intn(len(huaweiName))]
	}

}

func GenRandomUeType() string {
	ueTypeList := []string{"Macbook", "Lenovo", "Dell", "Huawei"}
	size := len(ueTypeList)
	return ueTypeList[rand.Intn(size)]
}
